package cn.zch.soft.util.wx.api;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import org.apache.http.HttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.util.EntityUtils;

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
/**
 * @author zchcpywin10
 */
public class MediaManager {
  public static Map<String, String> AppidToACCESS_TOKEN = new HashMap<>();
  public static String ACCESS_TOKEN = "";
  /**
   * 获取access_token
   */
  public static void getAccessToken(String appId, String appSecret) throws Exception {
    String result;
    String url = "https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid="
      //
      + appId + "&secret=" + appSecret;
    HttpGet get = new HttpGet(url);
    HttpResponse response = HttpClients.createDefault().execute(get);
    if (response.getStatusLine().getStatusCode() == 200) {
      result = EntityUtils.toString(response.getEntity());
      // JSONObject json = new JSONObject(Boolean.parseBoolean(result))
      // JSONObject jsonObject = new JSONObject()
      JSONObject jsonObject = JSON.parseObject(result);
      ACCESS_TOKEN = jsonObject.getString("access_token");
      AppidToACCESS_TOKEN.put(appId, ACCESS_TOKEN);
      // StringUtil.println("获取access_token OTHERWECHAT " + url + " " + ACCESS_TOKEN)
    }
  }
  public static JsonObject uploadFile(String appid, String secret, String filePath) throws Exception {
    //开始获取证书
    String accessToken = AppidToACCESS_TOKEN.get(appid);
    if (accessToken == null) {
      getAccessToken(appid, secret);
      accessToken = AppidToACCESS_TOKEN.get(appid);
    }
    // 上传文件请求路径
    String path = "https://api.weixin.qq.com/cgi-bin/material/add_material?access_token=" + accessToken + "&type=image";
    URL url = new URL(path);
    String result = null;
    File file = new File(filePath);
    if (!file.exists() || !file.isFile()) {
      throw new IOException("上传的文件不存在");
    }
    HttpURLConnection con = (HttpURLConnection) url.openConnection();
    con.setDoInput(true);
    con.setDoOutput(true);
    con.setUseCaches(false); // post方式不能使用缓存
    con.setRequestMethod("POST"); // 以Post方式提交表单，默认get方式
    // 设置请求头信息
    con.setRequestProperty("Connection", "Keep-Alive");
    con.setRequestProperty("Charset", "UTF-8");
    // 设置边界
    String BOUNDARY = "----------" + System.currentTimeMillis();
    con.setRequestProperty("Content-Type", "multipart/form-data; boundary=" + BOUNDARY);
    // 请求正文信息
    // 第一部分：
    StringBuilder sb = new StringBuilder();
    sb.append("--"); // 必须多两道线
    sb.append(BOUNDARY);
    sb.append("\r\n");
    sb.append("Content-Disposition: form-data;name=\"media\";filelength=\"" + file.length() + "\";filename=\"" + file.getName() + "\"\r\n");
    sb.append("Content-Type:application/octet-stream\r\n\r\n");
    System.out.println("sb==" + sb.toString());
    byte[] head = sb.toString().getBytes(StandardCharsets.UTF_8);
    // 获得输出流
    OutputStream out = new DataOutputStream(con.getOutputStream());
    // 输出表头
    out.write(head);
    // 文件正文部分
    // 把文件已流文件的方式 推入到url中
    DataInputStream in = new DataInputStream(new FileInputStream(file));
    int bytes = 0;
    byte[] bufferOut = new byte[1024];
    while ((bytes = in.read(bufferOut)) != -1) {
      out.write(bufferOut, 0, bytes);
    }
    in.close();
    // 结尾部分
    byte[] foot = ("\r\n--" + BOUNDARY + "--\r\n").getBytes(StandardCharsets.UTF_8);// 定义最后数据分隔线
    out.write(foot);
    out.flush();
    out.close();
    StringBuffer buffer = new StringBuffer();
    BufferedReader reader = null;
    try {
      // 定义BufferedReader输入流来读取URL的响应
      reader = new BufferedReader(new InputStreamReader(con.getInputStream()));
      String line = null;
      while ((line = reader.readLine()) != null) {
        buffer.append(line);
      }
      if (result == null) {
        result = buffer.toString();
      }
    }
    catch (IOException e) {
      System.out.println("发送POST请求出现异常！" + e);
      e.printStackTrace();
      throw new IOException("数据读取异常");
    }
    finally {
      if (reader != null) {
        reader.close();
      }
    }
    // JSONObject jsonObj = new JSONObject(result);
    // return jsonObj;
    JsonParser jsonparer = new JsonParser();// 初始化解析json格式的对象
    JsonObject json = jsonparer.parse(result).getAsJsonObject();
    System.out.println(json);
    return json;
  }
  public static void main(String[] args) throws Exception {
    String appId = "wx6ef0502a9f25ef9b";
    String appserect = "fbe276d1ba1582b077316908c2c16c45";
    String filePath = "E:\\1.png";
    JsonObject j = uploadFile(appId, appserect, filePath);
  }
}
